from envs import SchEnv
from agents import AgentMAPPO_SCH
from train import Config
from train import train_agent


# 指定环境和代理
env = SchEnv
agent = AgentMAPPO_SCH
env_args = {'env_name': "SchEnv",
                        'num_envs': 1,
                        'n_agents': 9,
                        'max_step': 4,
                        'state_dim': 4,
                        'action_dim': 2,
                        'if_discrete': None,
                        'gpu_id': int(0)}
# 指定超参数
args = Config(agent, env,env_args)
args.action_dim = 2
args.state_dim = 4
args.horizon_len = 1024

# 训练和评估
train_agent(args)
